PySpark transform multiple columns into a single column complex json